package com.example.springbootredis.config;

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.ClassPathResource;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.data.redis.serializer.GenericJackson2JsonRedisSerializer;
import org.springframework.data.redis.serializer.StringRedisSerializer;

import java.util.Set;

/**
 * @author: wukangrong
 * @date: 2021/3/1 11:37
 * @description:
 */
@Configuration
public class RedisConfig {
    @Bean
    public RedisTemplate<Object, Object> redisTemplate(RedisConnectionFactory redisConnectionFactory) {
        RedisTemplate<Object, Object> redisTemplate = new RedisTemplate<>();
        redisTemplate.setConnectionFactory(redisConnectionFactory);

        // 设置value的序列化规则和 key的序列化规则
        redisTemplate.setKeySerializer(new StringRedisSerializer());
        redisTemplate.setValueSerializer(new GenericJackson2JsonRedisSerializer());
        redisTemplate.setHashKeySerializer(new GenericJackson2JsonRedisSerializer());
        redisTemplate.setHashValueSerializer(new GenericJackson2JsonRedisSerializer());

        redisTemplate.afterPropertiesSet();
        return redisTemplate;
    }
    @Bean
    public DefaultRedisScript<Boolean> demoRedisScript(){
        DefaultRedisScript<Boolean> redisScript=new DefaultRedisScript<>();
        //设置lua脚本
        redisScript.setLocation(new ClassPathResource("demo.lua"));
        //设置返回类型
        redisScript.setResultType(java.lang.Boolean.class);
        return redisScript;
    }

    @Bean
    public DefaultRedisScript<Boolean> demo2RedisScript(){
        DefaultRedisScript<Boolean> redisScript=new DefaultRedisScript<>();
        //设置lua脚本
        redisScript.setLocation(new ClassPathResource("demo2.lua"));
        //设置返回类型
        redisScript.setResultType(Boolean.class);
        return redisScript;
    }

    @Bean
    public DefaultRedisScript<String> demo3RedisScript(){
        DefaultRedisScript<String> redisScript=new DefaultRedisScript<>();
        //设置lua脚本
        redisScript.setLocation(new ClassPathResource("demo3.lua"));
        //设置返回类型
        redisScript.setResultType(String.class);
        return redisScript;
    }

    @Bean
    public DefaultRedisScript<Boolean> limitRedisScript(){
        DefaultRedisScript<Boolean> redisScript=new DefaultRedisScript<>();
        //设置lua脚本
        redisScript.setLocation(new ClassPathResource("limit.lua"));
        //设置返回类型
        redisScript.setResultType(Boolean.class);
        return redisScript;
    }
}
